# -*- coding: utf-8 -*-
"""
pandas 均线计算

"""
import pandas as pd

#--------------------------------------------
# 分别计算5日、20日、60日的移动平均线
# 计算简单算术移动平均线MA - 注意：stock_data['close']为股票每天的收盘价
ma_list = [5, 20, 60]
for ma in ma_list:
    stock_data['MA_' + str(ma)] = pd.rolling_mean(stock_data['close'], ma)
    
# 计算指数平滑移动平均线EMA
for ma in ma_list:
    stock_data['EMA_' + str(ma)] = pd.ewma(stock_data['close'], span=ma)

# 将数据按照交易日期从近到远排序
stock_data.sort('date', ascending=False, inplace=True)
